"训练"
# Douzi
python train_douzi.py --config_file=configs/naic_round2_model_50_a.yml MODEL.DEVICE_ID "('0')" DATASETS.NAMES "('mydata')" \
DATASETS.ROOT_DIR "('../data')" MODEL.NAME "('resnet50_ibn_a')" MODEL.PRETRAIN_PATH "('../PreModel/r50_ibn_a.pth')" \
OUTPUT_DIR "('../save_model/')" \
SOLVER.MAX_EPOCHS 60 SOLVER.IMS_PER_BATCH 64
python train_douzi.py --config_file=configs/naic_round2_model_50_a.yml MODEL.DEVICE_ID "('0')" DATASETS.NAMES "('mydata')" \
DATASETS.ROOT_DIR "('../data')" MODEL.PRETRAIN_CHOICE "('finetune')" MODEL.NAME "('resnet50_ibn_a')" \
MODEL.PRETRAIN_PATH "('../save_model/resnet50_ibn_a/resnet50_ibn_a_60.pth')" \
OUTPUT_DIR "('../model_continue/')" \
SOLVER.MAX_EPOCHS 60 SOLVER.IMS_PER_BATCH 64
# RMGL
python RMGL_train.py --config_file=configs/RMGL_model_50_a.yml MODEL.DEVICE_ID "('0')" DATASETS.NAMES "('mydata')" \
DATASETS.ROOT_DIR "('../data')" MODEL.NAME "('resnet50_ibn_a')" MODEL.PRETRAIN_PATH "('../PreModel/r50_ibn_a.pth')" \
OUTPUT_DIR "('../save_model/')" \
SOLVER.MAX_EPOCHS 60 SOLVER.IMS_PER_BATCH 64
python RMGL_train.py --config_file=configs/RMGL_model_50_a.yml MODEL.DEVICE_ID "('0')" DATASETS.NAMES "('mydata')" \
DATASETS.ROOT_DIR "('../data')" MODEL.PRETRAIN_CHOICE "('finetune')" MODEL.NAME "('resnet50_ibn_a')" \
MODEL.PRETRAIN_PATH "('../save_model/resnet50_ibn_a/resnet50_ibn_a_60.pth')" \
OUTPUT_DIR "('../model_continue/')" \
SOLVER.MAX_EPOCHS 60 SOLVER.IMS_PER_BATCH 64

python RMGL_train.py --config_file=configs/RMGL_model_50_a.yml MODEL.DEVICE_ID "('0')" DATASETS.NAMES "('mydata')" \
DATASETS.ROOT_DIR "('../data')" MODEL.NAME "('resnet101_ibn_a')" MODEL.PRETRAIN_PATH "('../PreModel/resnet101_ibn_a-59ea0ac6.pth')" \
OUTPUT_DIR "('../save_model/')" \
SOLVER.MAX_EPOCHS 60 SOLVER.IMS_PER_BATCH 64
python RMGL_train.py --config_file=configs/RMGL_model_50_a.yml MODEL.DEVICE_ID "('0')" DATASETS.NAMES "('mydata')" \
DATASETS.ROOT_DIR "('../data')" MODEL.PRETRAIN_CHOICE "('finetune')" MODEL.NAME "('resnet101_ibn_a')" \
MODEL.PRETRAIN_PATH "('../save_model/resnet101_ibn_a/resnet101_ibn_a_60.pth')" \
OUTPUT_DIR "('../model_continue/')" \
SOLVER.MAX_EPOCHS 60 SOLVER.IMS_PER_BATCH 64

python RMGL_train.py --config_file=configs/RMGL_model_50_a.yml MODEL.DEVICE_ID "('0')" DATASETS.NAMES "('mydata')" \
DATASETS.ROOT_DIR "('../data')" MODEL.NAME "('se_resnet101_ibn_a')" MODEL.PRETRAIN_PATH "('../PreModel/se_resnet101_ibn_a-fabed4e2.pth')" \
OUTPUT_DIR "('../save_model/')" \
SOLVER.MAX_EPOCHS 60 SOLVER.IMS_PER_BATCH 64
python RMGL_train.py --config_file=configs/RMGL_model_50_a.yml MODEL.DEVICE_ID "('0')" DATASETS.NAMES "('mydata')" \
DATASETS.ROOT_DIR "('../data')" MODEL.NAME "('resnet101_ibn_b')" MODEL.PRETRAIN_PATH "('../PreModel/resnet101_ibn_b-c55f6dba.pth')" \
OUTPUT_DIR "('../save_model/')" \
SOLVER.MAX_EPOCHS 60 SOLVER.IMS_PER_BATCH 64
# SB
python SB_train.py --config_file=configs/RMGL_model_50_a.yml MODEL.DEVICE_ID "('0')" DATASETS.NAMES "('mydata')" \
DATASETS.ROOT_DIR "('../data')" MODEL.NAME "('resnet50_ibn_a')" MODEL.PRETRAIN_PATH "('../PreModel/r50_ibn_a.pth')" \
OUTPUT_DIR "('../save_model/')" \
SOLVER.MAX_EPOCHS 60 SOLVER.IMS_PER_BATCH 64
python SB_train.py --config_file=configs/RMGL_model_50_a.yml MODEL.DEVICE_ID "('0')" DATASETS.NAMES "('mydata')" \
DATASETS.ROOT_DIR "('../data')" MODEL.NAME "('resnet101_ibn_a')" MODEL.PRETRAIN_PATH "('../PreModel/resnet101_ibn_a-59ea0ac6.pth')" \
OUTPUT_DIR "('../save_model/')" \
SOLVER.MAX_EPOCHS 60 SOLVER.IMS_PER_BATCH 64
"测试"
"resnet50_ibn_a"
python test.py --config_file=configs/resnet50_ibn_a.yml MODEL.DEVICE_ID "('0')" DATASETS.NAMES "('mydata')" \
DATASETS.ROOT_DIR "('../data')" MODEL.NAME "('resnet50_ibn_a')" MODEL.PRETRAIN_CHOICE "('self')" \
TEST.WEIGHT "('../save_model/resnet50_ibn_a/resnet50_ibn_a_60.pth')" \
OUTPUT_DIR "('../json')" \
TEST.RE_RANKING True # 存放测试log
"resnet50_ibn_a pseudo"
python test.py --config_file=configs/naic_round2_model_50_a.yml MODEL.DEVICE_ID "('0')" DATASETS.NAMES "('mydata')" \
DATASETS.ROOT_DIR "('../data')" MODEL.NAME "('resnet50_ibn_a')" MODEL.PRETRAIN_CHOICE "('self')" \
TEST.WEIGHT "('../save_model/resnet50_ibn_a/resnet50_ibn_a_60.pth')" \
OUTPUT_DIR "('../json')" \
TEST.RE_RANKING True # 存放测试log
"efficientnet-b2"
python test.py --config_file=configs/naic_round2_model_50_a.yml MODEL.DEVICE_ID "('0')" DATASETS.NAMES "('mydata')" \
DATASETS.ROOT_DIR "('../data')" MODEL.NAME "('efficientnet-b2')" MODEL.PRETRAIN_CHOICE "('self')" \
TEST.WEIGHT "('../save_model/efficientnet-b2/efficientnet-b2_80.pth')" \
OUTPUT_DIR "('../json')" \
TEST.RE_RANKING True # 存放测试log
"hrnet"
python test.py --config_file=configs/naic_round2_model_50_a.yml MODEL.DEVICE_ID "('0')" DATASETS.NAMES "('mydata')" \
DATASETS.ROOT_DIR "('../data')" MODEL.NAME "('hrnet')" MODEL.PRETRAIN_CHOICE "('self')" \
TEST.WEIGHT "('../save_model/hrnet/hrnet_45.pth')" \
OUTPUT_DIR "('../json')" \
TEST.RE_RANKING True # 存放测试log
"RMGL resnet50_ibn_a"
python RMGL_test.py --config_file=configs/RMGL_model_50_a.yml MODEL.DEVICE_ID "('0')" DATASETS.NAMES "('mydata')" \
DATASETS.ROOT_DIR "('../data')" MODEL.NAME "('resnet50_ibn_a')" MODEL.PRETRAIN_CHOICE "('self')" \
TEST.WEIGHT "('../save_model/resnet50_ibn_a/resnet50_ibn_a_60.pth')" \
OUTPUT_DIR "('../json')" \
TEST.RE_RANKING True # 存放测试log
python RMGL_test.py --config_file=configs/RMGL_model_50_a.yml MODEL.DEVICE_ID "('0')" DATASETS.NAMES "('mydata')" \
DATASETS.ROOT_DIR "('../data')" MODEL.NAME "('resnet101_ibn_a')" MODEL.PRETRAIN_CHOICE "('self')" \
TEST.WEIGHT "('../save_model/resnet101_ibn_a/resnet101_ibn_a_60.pth')" \
OUTPUT_DIR "('../json')" \
TEST.RE_RANKING True # 存放测试log
python RMGL_test.py --config_file=configs/RMGL_model_50_a.yml MODEL.DEVICE_ID "('0')" DATASETS.NAMES "('mydata')" \
DATASETS.ROOT_DIR "('../data')" MODEL.NAME "('se_resnet101_ibn_a')" MODEL.PRETRAIN_CHOICE "('self')" \
TEST.WEIGHT "('../save_model/se_resnet101_ibn_a/se_resnet101_ibn_a_60.pth')" \
OUTPUT_DIR "('../json')" \
TEST.RE_RANKING True # 存放测试log
python RMGL_test.py --config_file=configs/RMGL_model_50_a.yml MODEL.DEVICE_ID "('0')" DATASETS.NAMES "('mydata')" \
DATASETS.ROOT_DIR "('../data')" MODEL.NAME "('resnet101_ibn_b')" MODEL.PRETRAIN_CHOICE "('self')" \
TEST.WEIGHT "('../save_model/resnet101_ibn_b/resnet101_ibn_b_60.pth')" \
OUTPUT_DIR "('../json')" \
TEST.RE_RANKING True # 存放测试log
"Douzi resnet50_ibn_a"
python test_douzi.py --config_file=configs/naic_round2_model_50_a.yml MODEL.DEVICE_ID "('0')" DATASETS.NAMES "('mydata')" \
DATASETS.ROOT_DIR "('../data')" MODEL.NAME "('resnet50_ibn_a')" MODEL.PRETRAIN_CHOICE "('self')" \
TEST.WEIGHT "('../save_model/resnet50_ibn_a/resnet50_ibn_a_60.pth')" \
OUTPUT_DIR "('../json')" \
TEST.RE_RANKING True # 存放测试log
"SB resnet50_ibn_a"
python SB_test.py --config_file=configs/RMGL_model_50_a.yml MODEL.DEVICE_ID "('0')" DATASETS.NAMES "('mydata')" \
DATASETS.ROOT_DIR "('../data')" MODEL.NAME "('resnet50_ibn_a')" MODEL.PRETRAIN_CHOICE "('self')" \
TEST.WEIGHT "('../save_model/resnet50_ibn_a/resnet50_ibn_a_60.pth')" \
OUTPUT_DIR "('../json')" \
TEST.RE_RANKING True # 存放测试log

"验证集验证ensemble效果"
python validate_ensemble.py

"测试集ensemble"
python MyEnsemble.py

python RMGL_test.py --config_file=configs/RMGL_model_50_a.yml MODEL.DEVICE_ID "('0')" DATASETS.NAMES "('mydata')" ^
DATASETS.ROOT_DIR "('../data')" MODEL.NAME "('resnet101_ibn_a')" MODEL.PRETRAIN_CHOICE "('self')" ^
TEST.WEIGHT "('../save_model/resnet101_ibn_a/resnet101_ibn_a_60.pth')" ^
OUTPUT_DIR "('../json')" ^
TEST.RE_RANKING True

python RMGL_test.py --config_file=configs/RMGL_model_50_a.yml MODEL.DEVICE_ID "('0')" DATASETS.NAMES "('mydata')" ^
DATASETS.ROOT_DIR "('../data')" MODEL.NAME "('resnet101_ibn_b')" MODEL.PRETRAIN_CHOICE "('self')" ^
TEST.WEIGHT "('../save_model/resnet101_ibn_b/resnet101_ibn_b_60.pth')" ^
OUTPUT_DIR "('../json')" ^
TEST.RE_RANKING True

python RMGL_test.py --config_file=configs/RMGL_model_50_a.yml MODEL.DEVICE_ID "('0')" DATASETS.NAMES "('mydata')" ^
DATASETS.ROOT_DIR "('../data')" MODEL.NAME "('se_resnet101_ibn_a')" MODEL.PRETRAIN_CHOICE "('self')" ^
TEST.WEIGHT "('../save_model/se_resnet101_ibn_a/se_resnet101_ibn_a_60.pth')" ^
OUTPUT_DIR "('../json')" ^
TEST.RE_RANKING True